how iterate through a list in python with index